from lenet import LeNet, LeNetSequetial
from lenet1 import MyNet
from torchsummary import summary

# net = LeNetSequetial(classes=2)
# net = LeNet(classes=2)
net = MyNet(classes=2)
net = net.cuda()
net.initialize_weights()

summary(net, input_size=(3, 112, 112))
print(net)